from .ACDC import get_acdc_loader, get_ssl_acdc_loader


def get_loader(args):
    if args.datasets == "acdc":
        label_loader, unlabel_loader, test_loader = get_ssl_acdc_loader(root=args.data_path, batch_size=args.batch_size, 
        unlabel_batch_size=args.unlabel_batch_size,label_num=args.label_num)
    else:
        raise NotImplementedError

    return label_loader, unlabel_loader, test_loader
